package com.taldh.algorithm.array;

import org.junit.Assert;
import org.junit.Test;

import java.util.Arrays;

public class ThreeSumClosestSolution {

    /**
     * 16. 最接近的三数之和
     *
     * 给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数，
     * 使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
     *
     * 例如，给定数组 nums = [-1，2，1，-4], 和 target = 1.
     *
     * 与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/3sum-closest
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     * @param nums
     * @param target
     * @return
     */
    public int threeSumClosest(int[] nums, int target) {
        if (nums == null || nums.length < 3) {
            return 0;
        }

        if (nums.length == 3) {
            return nums[0] + nums[1] + nums[2];
        }

        Arrays.sort(nums);
        int len = nums.length;
        int ans = Integer.MAX_VALUE / 2;
        for (int i = 0; i < len; i++) {
            int startIndex = i + 1;
            int endIndex = len - 1;
            while (startIndex < endIndex) {
                int sum = nums[i] + nums[startIndex] + nums[endIndex];
                if (Math.abs(target - sum) < Math.abs(target-ans)) {
                    ans = sum;
                }
                if (target > sum) {
                    startIndex ++;
                } else if (target < sum){
                    endIndex --;
                } else {
                    return ans;
                }
            }
        }
        return ans;
    }

    @Test
    public void test() {
        Assert.assertEquals(2, threeSumClosest(new int[]{1,2,-1,-4}, 1));
    }

    @Test
    public void test2() {
        Assert.assertEquals(1, threeSumClosest(new int[]{-1,1,1}, 0));
    }

    @Test
    public void test3() {
        Assert.assertEquals(0, threeSumClosest(new int[]{0,2,1,-3}, 1));
    }

    @Test
    public void test4() {
        Assert.assertEquals(-1, threeSumClosest(new int[]{1,1,-1,-1,3}, -1));
    }

}